package main

import (
	"flag"
	"fmt"
	"os"
)

var (
	baseDir string
	filename string
	downloadColumn int
)

func init () {
	flag.StringVar(&baseDir, "basedir", "storage/image", "图片保存的位置")
	flag.StringVar(&filename, "filename", "", "读取的csv文件名")
	flag.IntVar(&downloadColumn, "downloadColumn", 0, "需要下载的是csv的第几列")
	flag.Parse()

	// 判断参数是否有问题
	if len(filename) ==0 {
		fmt.Println("请输入文件名")
		os.Exit(0)
	}
	// 检查基础文件夹是否存在，不存在则创建
	flag, err := PathExists(baseDir)
	if err != nil {
		fmt.Println("Error: ", err)
		os.Exit(-1)
	}
	if flag == false {
		err := os.MkdirAll(baseDir, os.ModePerm)
		if err != nil {
			fmt.Println("Error: ", err)
			os.Exit(-1)
		}
	}
}

func main() {
	csvDownload := NewCsvDownload()
	csvDownload.SetBaseDir(baseDir).SetCsvFile(filename).SetDownloadColumn(downloadColumn).Download()
}
